//给你一个下标从 0 开始的整数数组 nums ，以及整数 modulo 和整数 k 。 
//
// 请你找出并统计数组中 趣味子数组 的数目。 
//
// 如果 子数组 nums[l..r] 满足下述条件，则称其为 趣味子数组 ： 
//
// 
// 在范围 [l, r] 内，设 cnt 为满足 nums[i] % modulo == k 的索引 i 的数量。并且 cnt % modulo == k 。
// 
// 
//
// 以整数形式表示并返回趣味子数组的数目。 
//
// 注意：子数组是数组中的一个连续非空的元素序列。 
//
// 
//
// 示例 1： 
//
// 
//输入：nums = [3,2,4], modulo = 2, k = 1
//输出：3
//解释：在这个示例中，趣味子数组分别是： 
//子数组 nums[0..0] ，也就是 [3] 。 
//- 在范围 [0, 0] 内，只存在 1 个下标 i = 0 满足 nums[i] % modulo == k 。
//- 因此 cnt = 1 ，且 cnt % modulo == k 。
//子数组 nums[0..1] ，也就是 [3,2] 。
//- 在范围 [0, 1] 内，只存在 1 个下标 i = 0 满足 nums[i] % modulo == k 。
//- 因此 cnt = 1 ，且 cnt % modulo == k 。
//子数组 nums[0..2] ，也就是 [3,2,4] 。
//- 在范围 [0, 2] 内，只存在 1 个下标 i = 0 满足 nums[i] % modulo == k 。
//- 因此 cnt = 1 ，且 cnt % modulo == k 。
//可以证明不存在其他趣味子数组。因此，答案为 3 。 
//
// 示例 2： 
//
// 
//输入：nums = [3,1,9,6], modulo = 3, k = 0
//输出：2
//解释：在这个示例中，趣味子数组分别是： 
//子数组 nums[0..3] ，也就是 [3,1,9,6] 。
//- 在范围 [0, 3] 内，只存在 3 个下标 i = 0, 2, 3 满足 nums[i] % modulo == k 。
//- 因此 cnt = 3 ，且 cnt % modulo == k 。
//子数组 nums[1..1] ，也就是 [1] 。
//- 在范围 [1, 1] 内，不存在下标满足 nums[i] % modulo == k 。
//- 因此 cnt = 0 ，且 cnt % modulo == k 。
//可以证明不存在其他趣味子数组，因此答案为 2 。 
//
// 
//
// 提示： 
//
// 
// 1 <= nums.length <= 105 
// 1 <= nums[i] <= 10⁹ 
// 1 <= modulo <= 10⁹ 
// 0 <= k < modulo 
// 
//
// Related Topics 数组 哈希表 前缀和 👍 58 👎 0


package LeetCode.editor.cn;


import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author ldltd
 * @date 2025-04-25 10:23:11
 * @description 2845.统计趣味子数组的数目
 
 */
 
public class CountOfInterestingSubarrays {
    public static void main(String[] args) {
    //测试代码
    CountOfInterestingSubarrays fun = new CountOfInterestingSubarrays();
    Solution solution= fun.new Solution();
    
    }

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
        //定义前缀和prefix，prefix表示前i个模modulo的等于k的个数再modulo的值
    //用一个map统计出现的次数
    //对于每个prefix，查找map中（prefix-k)%modulo的计数
    public long countInterestingSubarrays(List<Integer> nums, int modulo, int k) {
        Map<Integer, Integer> countMap = new HashMap<>();
        countMap.put(0, 1); // 初始状态，前缀和为0出现1次
        long prefix = 0;
        long res = 0;

        for (int num : nums) {
            if (num % modulo == k) {
                prefix = (prefix + 1) % modulo;
            }
            // 我们要找 (prefix - k) mod modulo  防止负数
            int target = (int)((prefix - k + modulo) % modulo);
            res += countMap.getOrDefault(target, 0);
            countMap.put((int)prefix, countMap.getOrDefault((int)prefix, 0) + 1);
        }

        return res;
    }
}
//leetcode submit region end(Prohibit modification and deletion)

}
